/*******************************************************************************
 * Copyright (c) 2006 CEA List.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     CEA List - initial API and implementation
 *******************************************************************************/
package com.cea.papyrus.diagram.classdiagram.helper;

import java.util.ArrayList;

import org.eclipse.draw2d.geometry.Dimension;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.jface.preference.IPreferenceStore;
import org.eclipse.swt.graphics.FontData;
import org.eclipse.uml2.uml.Classifier;
import org.eclipse.uml2.uml.DataType;
import org.eclipse.uml2.uml.Element;
import org.eclipse.uml2.uml.Enumeration;
import org.eclipse.uml2.uml.Interface;
import org.eclipse.uml2.uml.Operation;

import com.cea.papyrus.core.PapyrusPlugin;
import com.cea.papyrus.core.helper.NodeHelper;
import com.cea.papyrus.core.preference.ClassDiagramCommonPreferenceConstants;
import com.cea.papyrus.core.utils.Debug;
import com.cea.papyrus.diagraminterchange2.di2.Diagram;
import com.cea.papyrus.diagraminterchange2.di2.GraphElement;
import com.cea.papyrus.diagraminterchange2.di2.GraphNode;
import com.cea.papyrus.diagraminterchange2.di2.Uml1SemanticModelBridge;

/**
 * Factory that can create and delete an {@link Operation}}.
 * 
 * @author Patrick Tessier, Remi Schnekenburger
 */
public class NodeOperationHelper extends NodeHelper {
	
	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.NodeFactory#createGraphElement(org.eclipse.uml2.uml.Element, com.cea.papyrus.diagraminterchange2.di2.GraphElement, org.eclipse.draw2d.geometry.Point, org.eclipse.draw2d.geometry.Dimension)
	 */
	/**
	 * 
	 * 
	 * @param umlChild 
	 * @param size 
	 * @param location 
	 * @param parent 
	 * 
	 * @return 
	 */
	@Override
	public GraphNode createGraphElement(Element umlChild, GraphElement parent, Point location, Dimension size) {
		Debug.debug(this,"Call NodeOperationFactory.createGraphElement()", channel); //$NON-NLS-1$

		// creates the graph element corresponding to the Operation
		GraphNode graphNode = super.createGraphElement(umlChild, parent, location, size);
		
		// finally, adds the Operation graph element to its parent graph element
		parent.getContained().add(graphNode);

		return graphNode;
	}

	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.INodeFactory#createGraphNode(org.eclipse.uml2.uml.NamedElement, com.cea.papyrus.diagraminterchange2.di2.Diagram)
	 */
	/**
	 * 
	 * 
	 * @param umlElement 
	 * @param diagram 
	 * 
	 * @return 
	 */
	@Override
	public ArrayList createGraphNode(Element umlElement, Diagram diagram) {
		return new ArrayList();
	}
	
	/**
	 * Returns the label of an operation.
	 * 
	 * @param o the operation to calculate
	 * 
	 * @return the label of this operation
	 * 
	 * @deprecated use {@link com.cea.utils.Operation#getLabel(int) }instead
	 */
	@Deprecated
	public String getString(Operation o) {
		return getString(new com.cea.papyrus.umlutils.Operation(o));
	}
	
	/**
	 * Returns the label of an operation.
	 * 
	 * @param o the operation to calculate
	 * 
	 * @return the label of this operation
	 */
	public String getString(com.cea.papyrus.umlutils.Operation o){
		return o.getLabel();
	}
	
	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.NodeFactory#canExecuteCreateCommand(com.cea.papyrus.diagraminterchange2.di2.GraphNode)
	 */
	/**
	 * 
	 * 
	 * @param parent 
	 * 
	 * @return 
	 */
	@Override
	public boolean canCreateGraphElement(GraphElement parent){

		if (!(parent instanceof Diagram)) {//the parent must be a graphNode

			if(parent.getSemanticModel()==null){ // the parent must a Feature
				GraphElement gp=parent.getContainer();//look for GreatParent

				if (gp!=null){
					if (!(gp instanceof Diagram)) {//great parent is not a diagram
						if(gp.getSemanticModel()!=null){ //it is linked to an UML element
							Element elt= ((Uml1SemanticModelBridge)gp.getSemanticModel()).getElement();

							if ((elt instanceof Classifier)&& (!(elt instanceof Enumeration))&& (gp.getContained().get(1)==parent)){ //this is the first container of a classifier.
								return true;
							}
						}
					}
				}
			}
		}
		return false;
	}
	
	///////////////////////////////////////////////////////////////////////////
	// UML Management
	///////////////////////////////////////////////////////////////////////////
	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.NodeFactory#createElement(java.lang.String, org.eclipse.uml2.uml.umlElement)
	 */
	/**
	 * 
	 * 
	 * @param umlParent 
	 * @param name 
	 * 
	 * @return 
	 */
	@Override
	public org.eclipse.uml2.uml.Element createUMLElement(String name, org.eclipse.uml2.uml.Element umlParent){
		Debug.debug(this,"Call NodeOperationFactory.createElement()", channel); //$NON-NLS-1$
		org.eclipse.uml2.uml.Operation operation=null;
		if (umlParent instanceof org.eclipse.uml2.uml.Class){
			operation=((org.eclipse.uml2.uml.Class)umlParent).createOwnedOperation(name,null,null);
		}
		if (umlParent instanceof org.eclipse.uml2.uml.Interface){
			operation=((org.eclipse.uml2.uml.Interface)umlParent).createOwnedOperation(name,null,null);
		}
		if (umlParent instanceof org.eclipse.uml2.uml.DataType){
			operation=((org.eclipse.uml2.uml.DataType)umlParent).createOwnedOperation(name,null,null);
		}
		return operation;
	}

	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.NodeFactory#canExecuteCreateElement(org.eclipse.uml2.uml.Element)
	 */
	/**
	 * 
	 * 
	 * @param umlParent 
	 * 
	 * @return 
	 */
	@Override
	public boolean canCreateUMLElement(Element umlParent) {
		if(umlParent instanceof org.eclipse.uml2.uml.Class || umlParent instanceof Interface|| umlParent instanceof DataType) {
			return true;
		}
		return false;
	}
	
	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.NodeFactory#addElement(org.eclipse.uml2.uml.Element, org.eclipse.uml2.uml.Element)
	 */
	/**
	 * 
	 * 
	 * @param element 
	 * @param umlParent 
	 * 
	 * @return 
	 */
	@Override
	public boolean addUMLElement(Element element, Element umlParent) {
		if(umlParent instanceof org.eclipse.uml2.uml.Class) {
			return ((org.eclipse.uml2.uml.Class)umlParent).getOwnedOperations().add((Operation)element);
		}
		if(umlParent instanceof Interface) {
			return ((Interface)umlParent).getOwnedOperations().add((Operation)element);
		}
		return false;
	}
	
	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.NodeFactory#removeElement(org.eclipse.uml2.uml.Element, org.eclipse.uml2.uml.Element)
	 */
	/**
	 * 
	 * 
	 * @param element 
	 * @param umlParent 
	 * 
	 * @return 
	 */
	@Override
	public boolean removeUMLElement(Element element, Element umlParent) {
		if(umlParent instanceof org.eclipse.uml2.uml.Class) {
			return ((org.eclipse.uml2.uml.Class)umlParent).getOwnedOperations().remove(element);
		}
		if(umlParent instanceof Interface) {
			return ((Interface)umlParent).getOwnedOperations().remove(element);
		}
		if(umlParent instanceof DataType) {
			return ((DataType)umlParent).getOwnedOperations().remove(element);
		}
		return false;
	}
	
	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.factories.NodeFactory#getBaseString()
	 */
	/**
	 * 
	 * 
	 * @return 
	 */
	@Override
	public String getBaseString() {
		return "Operation_"; //$NON-NLS-1$
	}

	/* (non-Javadoc)
	 * @see com.cea.papyrus.common.helper.NodeHelper#setGraphElementPreferences(com.cea.papyrus.diagraminterchange2.di2.GraphElement)
	 */
	/**
	 * 
	 * 
	 * @param graphElement 
	 */
	@Override
	public void setGraphElementPreferences(GraphElement graphElement) {
		// Get the preference store
		IPreferenceStore store = PapyrusPlugin.getDefault().getPreferenceStore();
		
		// Set colors and font
		graphElement.setBackgroundColor(ClassDiagramCommonPreferenceConstants.getPreferenceOperationBackgroundColor(store));
		graphElement.setBorderColor(ClassDiagramCommonPreferenceConstants.getPreferenceOperationBorderColor(store));
		graphElement.setFontColor(ClassDiagramCommonPreferenceConstants.getPreferenceOperationFontColor(store));
		FontData[] fontdata = ClassDiagramCommonPreferenceConstants.getPreferenceOperationFont(store);
		if (fontdata.length>0){
			graphElement.setFontFamily(fontdata[0].getName());
			graphElement.setFontSize(fontdata[0].getHeight());
		}
	}
}